#!/bin/bash

if [ `whoami` != "root" ]; then
    echo "需要以root用户执行"
    exit -1;
fi

echo "请注意安装包所在的路径不要包含中文 中文路径安装会出错"
echo "开始安装工具包 --------------"


toolkit=~/Ascend-cann-toolkit_8.0.0_linux-aarch64.run
toolbox=~/Ascend-mindx-toolbox_7.0.T10_linux-aarch64.run
# 判断文件是否存在
if [ ! -f $toolkit ]; then
    echo $toolkit 文件不存在-需要指定文件所在目录
    exit -1;
fi
# 判断文件是否存在
if [ ! -f $toolbox ]; then
    echo $toolbox 文件不存在-需要指定文件所在目录
    exit -1;
fi


chmod +x $toolkit
chmod +x $toolbox
rm -r /usr/local/Ascend/ascend-toolkit/
rm -r /usr/local/Ascend/toolbox/
$toolkit --install --quiet
$toolbox --install --quiet
touch /root/.bashrc

echo "#tool_begin" >> ~/.bashrc
echo '. /usr/local/Ascend/ascend-toolkit/set_env.sh' >> /root/.bashrc
echo '. /usr/local/Ascend/toolbox/set_env.sh' >> /root/.bashrc
echo "#tool_end" >> ~/.bashrc
source ~/.bashrc

echo "toolkit toolbox 安装完成 --------------"
